python program for data structure and algorithms